"""
ihrm 人类资源管理系统 api 文件
"""
import requests
import app


# 登录
class LoginAPI:
    def __init__(self):
        self.login_url = app.BASE_URL + '/api/sys/login'
        self.headers = app.HEADERS

    def login(self, json_data):
        return requests.post(url=self.login_url, headers=self.headers, json=json_data)


# 员工管理
class EmpAPI:
    def __init__(self):
        self.emp_url = app.BASE_URL + '/api/sys/user'

    # 添加员工
    def add_emp(self, headers, json_data):
        return requests.post(url=self.emp_url, headers=headers, json=json_data)

    # 查询员工
    def query_emp(self, emp_id, headers):
        return requests.get(url=self.emp_url + '/' + emp_id, headers=headers)

    # 修改员工
    def modify_emp(self, emp_id, headers, json_data):
        return requests.put(url=self.emp_url + '/' + emp_id, headers=headers, json=json_data)

    # 删除员工
    def del_emp(self, emp_id, headers):
        return requests.delete(url=self.emp_url + '/' + emp_id, headers=headers)

    # 查询员工列表：
    def query_emp_list(self, headers):
        return requests.get(url=self.emp_url + '?page=1&size=10', headers=headers)
